package com.mango.leet.code.week1.game292;

/**
 * 6058. 统计打字方案数
 * @Author: mango
 * @Date: 2022/5/8 10:56 上午
 */

import com.google.common.base.Stopwatch;

import java.util.concurrent.TimeUnit;

/**
 * Alice 在给 Bob 用手机打字。数字到字母的 对应 如下图所示。
 *
 *
 *
 * 为了 打出 一个字母，Alice 需要 按 对应字母 i 次，i 是该字母在这个按键上所处的位置。
 *
 * 比方说，为了按出字母 's' ，Alice 需要按 '7' 四次。类似的， Alice 需要按 '5' 两次得到字母  'k' 。
 * 注意，数字 '0' 和 '1' 不映射到任何字母，所以 Alice 不 使用它们。
 * 但是，由于传输的错误，Bob 没有收到 Alice 打字的字母信息，反而收到了 按键的字符串信息 。
 *
 * 比方说，Alice 发出的信息为 "bob" ，Bob 将收到字符串 "2266622" 。
 * 给你一个字符串 pressedKeys ，表示 Bob 收到的字符串，请你返回 Alice 总共可能发出多少种文字信息 。
 *
 * 由于答案可能很大，将它对 109 + 7 取余 后返回。
 *
 *
 *
 * 示例 1：
 *
 * 输入：pressedKeys = "22233"
 * 输出：8
 * 解释：
 * Alice 可能发出的文字信息包括：
 * "aaadd", "abdd", "badd", "cdd", "aaae", "abe", "bae" 和 "ce" 。
 * 由于总共有 8 种可能的信息，所以我们返回 8 。
 * 示例 2：
 *
 * 输入：pressedKeys = "222222222222222222222222222222222222"
 * 输出：82876089
 * 解释：
 * 总共有 2082876103 种 Alice 可能发出的文字信息。
 * 由于我们需要将答案对 109 + 7 取余，所以我们返回 2082876103 % (109 + 7) = 82876089 。
 *
 *
 * 提示：
 *
 * 1 <= pressedKeys.length <= 105
 * pressedKeys 只包含数字 '2' 到 '9' 。
 */
public class LC6058 {
    public static void main(String[] args) {
        //String pressedKeys = "22233";
        //String pressedKeys = "456";
        //String pressedKeys = "7777";// pqrs: pppp, pqp, pr, s, qpp, qq, rp
        Stopwatch sw = Stopwatch.createStarted();
        String pressedKeys = "222222222222222222222222222222222222";
        System.out.println(new Solution().countTexts(pressedKeys) + " 耗时:"+sw.elapsed(TimeUnit.SECONDS)+"s");
    }
    static class Solution {
        // 暴力递归
        public int countTexts(String pressedKeys) {
            if (pressedKeys == null || pressedKeys.length() == 0) {
                return 0;
            }
            return process(pressedKeys.toCharArray(), 0).intValue();
        }

        private Long process(char[] arr, int i) {
            if (i >= arr.length - 1) {
                return 1L;
            }
            long result = 1L;
            if (arr[i] == '7' || arr[i] == '9') {
                if (i + 3 < arr.length && arr[i] == arr[i + 1] && arr[i] == arr[i + 2] && arr[i] == arr[i + 3]) {
                    result = process(arr, i + 1) +
                            process(arr, i + 2) +
                            process(arr, i + 3) +
                            process(arr, i + 4);
                }
            }
            if(i+2 < arr.length && arr[i] == arr[i+1] && arr[i] == arr[i+2]){
                result = process(arr, i + 1) +
                        process(arr, i + 2) +
                        process(arr, i + 3);
            }else if(i+1 < arr.length && arr[i] == arr[i+1]){
                result = process(arr, i + 1) +
                        process(arr, i + 2);
            }else{
                result = process(arr, i + 1);
            }
            return (long) (result % (Math.pow(10,9)+7));
        }
    }
}
